python calculating with functions